from app.errors import errors_bp
from flask import render_template
from app.ext import db

@errors_bp.app_errorhandler(404)
def not_found_error(error):
    return render_template('errors/404.html'), 404


@errors_bp.app_errorhandler(500)
def internal_error(error):
    db.session.rollback()
    return render_template('errors/500.html'), 500

# 添加全局401无权限页面
@errors_bp.app_errorhandler(401)
def unauthorized_access(error):
    return render_template('errors/401.html'), 401